import E_command
import Elysium_BD
import Elysium_XF
import time


def main():
    time.sleep(0.01)
    # 加载程序信息
    E_command.command_inf()
    while True:
        # 显示模型选择菜单
        E_command.command_code("请选择要使用的模型：")
        # print("0. 程序信息")
        print("1. Elysium_BD")
        print("2. Elysium_XF")

        # 获取用户选择
        choice = input("请输入选项编号（输入'exit'退出）：")

        # 处理用户选择
        if choice == "1":
            Elysium_BD.Elysium_BD()
        elif choice == "2":
            Elysium_XF.Elysium_XF()
        elif choice.lower() == "exit":
            E_command.command_code("进程结束")
            break
        else:
            E_command.command_warn("无效选项")

if __name__ == '__main__':
    main()
